package com.heima.kafka.sample;

/**
 * @Author yuan
 * @Date 2023/1/30
 * @ApiNote
 */

import org.apache.kafka.clients.consumer.ConsumerConfig;
import org.apache.kafka.clients.consumer.ConsumerRecord;
import org.apache.kafka.clients.consumer.ConsumerRecords;
import org.apache.kafka.clients.consumer.KafkaConsumer;

import java.time.Duration;
import java.util.Collections;
import java.util.Properties;

/**
 * 消费者
 */
public class ConsumerQuickStart {
    public static void main(String[] args) {

        // 1. kafka链接配置信息
        Properties prop = new Properties();
        // 1.1 链接地址
        prop.put(ConsumerConfig.BOOTSTRAP_SERVERS_CONFIG, "192.168.200.130:9092");
        // 1.2 key value 数据序列化
        prop.put(ConsumerConfig.KEY_DESERIALIZER_CLASS_CONFIG, "org.apache.kafka.common.serialization.StringDeserializer");
        prop.put(ConsumerConfig.VALUE_DESERIALIZER_CLASS_CONFIG, "org.apache.kafka.common.serialization.StringDeserializer");
        // 1.3 设置消费者组
        // prop.put(ConsumerConfig.GROUP_ID_CONFIG, "group1");
        prop.put(ConsumerConfig.GROUP_ID_CONFIG, "group2");

        // 2. 创建消费者对象
        KafkaConsumer<String, String> consumer = new KafkaConsumer<String, String>(prop);

        // 3. 订阅主题
        //consumer.subscribe(Collections.singletonList("topic-first"));

        // 流式计算
        consumer.subscribe(Collections.singletonList("itcast-topic-out"));

        // 4. 拉取消息
        /*while (true) {
            ConsumerRecords<String, String> consumerRecords = consumer.poll(Duration.ofMillis(1000));
            for (ConsumerRecord<String, String> consumerRecord : consumerRecords) {
                System.out.println(consumerRecord.key());
                System.out.println(consumerRecord.value());
                // 分区号
                System.out.println(consumerRecord.partition());
                // 偏移量
                System.out.println(consumerRecord.offset());

                // 同步提交偏移量
                *//*try {
                    consumer.commitSync();
                }catch (Exception e) {
                    System.out.println("记录失败提交的异常: " + e);
                }*//*
            }

            // 异步提交偏移量
            *//*consumer.commitAsync(new OffsetCommitCallback() {
                @Override
                public void onComplete(Map<TopicPartition, OffsetAndMetadata> map, Exception e) {
                    if(e != null) {
                        System.out.println("记录错误的提交偏移量: " + map + ", 错误信息: " + e);
                    }
                }
            });*//*
        }*/

        // 同步和异步提交偏移量
        try {
            while (true) {
                ConsumerRecords<String, String> consumerRecords = consumer.poll(Duration.ofMillis(1000));
                for (ConsumerRecord<String, String> consumerRecord : consumerRecords) {
                    System.out.println(consumerRecord.key());
                    System.out.println(consumerRecord.value());
                    // 分区号
                    System.out.println(consumerRecord.partition());
                    // 偏移量
                    System.out.println(consumerRecord.offset());
                }

                // 异步提交偏移量
                consumer.commitAsync();
            }
        }catch (Exception e) {
            e.printStackTrace();
            System.out.println("记录错误信息: " + e);
        }finally {
            // 同步
            consumer.commitSync();
        }
    }
}
